@echo off
chcp 65001 >nul
title AI Booster 部署工具

echo.
echo ========================================
echo     AI Booster 一键部署工具
echo ========================================
echo.

:MAIN_MENU
echo 请选择操作:
echo.
echo 1. 部署到单个服务器
echo 2. 批量部署到多台服务器
echo 3. 创建服务器列表模板
echo 4. 检查环境
echo 5. 查看帮助
echo 6. 退出
echo.
set /p choice=请输入选项 (1-6): 

if "%choice%"=="1" goto SINGLE_DEPLOY
if "%choice%"=="2" goto BATCH_DEPLOY
if "%choice%"=="3" goto CREATE_TEMPLATE
if "%choice%"=="4" goto CHECK_ENV
if "%choice%"=="5" goto SHOW_HELP
if "%choice%"=="6" goto EXIT
goto MAIN_MENU

:SINGLE_DEPLOY
echo.
echo ========================================
echo         单服务器部署
echo ========================================
set /p server_ip=请输入服务器IP地址: 
set /p username=请输入SSH用户名 (默认为root): 
if "%username%"=="" set username=root
set /p port=请输入SSH端口 (默认为22): 
if "%port%"=="" set port=22

echo.
echo 正在部署到 %username%@%server_ip%:%port%...
powershell.exe -ExecutionPolicy Bypass -File "deploy.ps1" -ServerIP "%server_ip%" -Username "%username%" -Port %port%

echo.
pause
goto MAIN_MENU

:BATCH_DEPLOY
echo.
echo ========================================
echo         批量部署
echo ========================================
echo.
if not exist "servers.txt" (
    echo 错误: 找不到 servers.txt 文件
    echo 请先选择选项3创建服务器列表模板
    echo.
    pause
    goto MAIN_MENU
)

echo 正在读取 servers.txt 进行批量部署...
powershell.exe -ExecutionPolicy Bypass -File "deploy.ps1" -Command "batch"

echo.
pause
goto MAIN_MENU

:CREATE_TEMPLATE
echo.
echo ========================================
echo       创建服务器列表模板
echo ========================================
echo.
powershell.exe -ExecutionPolicy Bypass -File "deploy.ps1" -Command "create-template"
echo.
echo 模板创建完成！请编辑 servers.txt 文件添加您的服务器信息
echo 格式: IP:用户名:端口
echo 示例: 192.168.1.100:root:22
echo.
set /p edit_file=是否现在打开文件进行编辑? (y/n): 
if /i "%edit_file%"=="y" notepad servers.txt

echo.
pause
goto MAIN_MENU

:CHECK_ENV
echo.
echo ========================================
echo         环境检查
echo ========================================
echo.
powershell.exe -ExecutionPolicy Bypass -File "deploy.ps1" -Command "check"
echo.
pause
goto MAIN_MENU

:SHOW_HELP
echo.
echo ========================================
echo         使用帮助
echo ========================================
echo.
echo 文件说明:
echo   install_ai_booster.sh  - Linux安装脚本
echo   manage_ai_booster.sh   - 配置管理脚本
echo   deploy.ps1             - PowerShell部署脚本
echo   deploy.bat             - 本批处理文件
echo   README.md              - 详细说明文档
echo   QUICK_START.md         - 快速开始指南
echo.
echo 使用流程:
echo   1. 选择"部署到单个服务器"或"批量部署"
echo   2. 输入服务器信息进行部署
echo   3. SSH登录到目标服务器
echo   4. 运行: sudo bash /tmp/install_ai_booster.sh
echo   5. 使用: source /etc/ai_booster enable
echo.
echo 注意事项:
echo   - 需要安装SSH客户端 (OpenSSH)
echo   - 建议配置SSH密钥认证
echo   - 确保目标服务器可通过SSH访问
echo.
pause
goto MAIN_MENU

:EXIT
echo.
echo 感谢使用 AI Booster 部署工具！
echo.
pause
exit /b 0
